import unittest
import class_test_demo
import class_test_demo1


suite = unittest.TestSuite()
loader = unittest.TestLoader()
# 从模块中加载
suite1 = loader.loadTestsFromModule(class_test_demo)
suite2 = loader.loadTestsFromModule(class_test_demo1)

all_suite = unittest.TestSuite([suite1, suite2])
if __name__ == "__main__":
    unittest.main(defaultTest="all_suite")
